package br.ufpb.di.knowledgetv.instancias;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;

import br.ufpb.di.knowledgetv.util.Util;

public class Distribuicoes {

	private ArrayList<Distribuicao> distribuicoes;

	public Distribuicoes(double[] dist) {
		distribuicoes = new ArrayList<Distribuicao>();
		
		for(int i = 0; i < dist.length; i++)
			distribuicoes.add( new Distribuicao( Util.classes[i] , dist[i]) );

		Collections.sort(distribuicoes, new Comparator<Distribuicao>() {
			public int compare(Distribuicao o1, Distribuicao o2) {
				if (o1.getProbabilidade() < o2.getProbabilidade())
					return 1;
				if (o1.getProbabilidade() == o2.getProbabilidade())
					return 0;
				return -1;
			}
		});
	}

	public ArrayList<Distribuicao> getDistribuicoes() {
		return distribuicoes;
	}

}
